from django.test import TestCase, Client
from main.models import Penalty


class MainTest(TestCase):
	def test_login(self):
		response = self.client.get('/login/')
		self.assertEqual(response.status_code, 200)

	def test_p(self):
		c = Client()
		c.login(username='mod', password='1')
		response = c.get('/p/1')
		self.assertEqual(response.status_code, 200)